# 导航控制

import streamlit as st
from streamlit_model.streamlit_utils.session_utils import init_session

init_session()


def create_navigation():
    """创建顶部导航标签"""
    return st.radio(
        "选择测试类型:",
        ["手动输入需求描述生成用例", "根据需求文档按照要求生成用例", "上传图片和要求生成用例", "已生成的用例"],
        # 是否水平排列
        horizontal=True,
        # 默认选中第 0 个选项
        index=0,
        # 格式化函数
        format_func=str,
        help="只能选一个哦",
        label_visibility="collapsed"

    )


# 创建接口测试用例页面导航
def create_navigation_api():
    """创建顶部导航标签"""
    return st.radio(
        "选择测试类型:",
        ["根据接口文档按照要求生成用例", "已生成的用例"],
        # 是否水平排列
        horizontal=True,
        # 默认选中第 0 个选项
        index=0,
        # 格式化函数
        format_func=str,
        help="只能选一个哦",
        label_visibility="collapsed"

    )
